#!/bin/bash

if [[ $# -ne 2 ]]; then
    echo "The format is as follows:"
    echo "bash openblas-0.3.21.sh buildpath installpath"
    echo ":param buildpath: 应用构建绝对路径"
    echo ":param installpath: 应用安装绝对路径"
    exit 1
fi

[[ ! "$1" =~ ^/.* || ! -d "$1" || "$1" = "/" ]] &&  echo "请输入正确的构建路径" && exit 1
[[ ! "$2" =~ ^/.* || ! -d "$2" || "$2" = "/" ]] &&  echo "请输入正确的安装路径" && exit 1
if [[ "$1" == "$2" ]]; then echo "构建路径与安装路径不能相同，请输入正确的路径"; exit 1; fi

buildpath=$1
installpath=$2
[[ "$buildpath" =~ .*/$ ]] && buildpath=${buildpath%/*}
[[ "$installpath" =~ .*/$ ]] && installpath=${installpath%/*}
set -e

download_pkg()
{
    wget -t 40 -c $1 -O $buildpath/OpenBLAS-0.3.21.tar.gz
    tar xf $buildpath/OpenBLAS-0.3.21.tar.gz -C $buildpath
    #rm -rf $buildpath/OpenBLAS-0.3.21.tar.gz
}

download_pkg https://ghproxy.com/github.com/xianyi/OpenBLAS/archive/refs/tags/v0.3.21.tar.gz
cd $buildpath/OpenBLAS-0.3.21
CC=gcc CXX=g++ FC="gfortran -fallow-argument-mismatch" make
make PREFIX=$installpath/openblas-0.3.21 install

cat>"$installpath/openblas-0.3.21/openblas_modulefiles"<<EOF
#%Module1.0
conflict openblas
variable modfile [file normalize [info script]]
proc getModulefileDir {} {
    variable modfile
    set modfile_path [file dirname \$modfile]
    return \$modfile_path
}
set openblas [getModulefileDir]
setenv OPENBLAS \$openblas
prepend-path LD_LIBRARY_PATH \$openblas/lib
prepend-path INCLUDE \$openblas/include
EOF

#rm -rf $buildpath/OpenBLAS-0.3.21
